import os
from dotenv import load_dotenv
from py2neo import Graph, NodeMatcher, RelationshipMatcher

# 加载 .env 文件中的环境变量
load_dotenv()
# 使用 os.getenv 获取环境变量的值
database_url = os.getenv("DATABASE_URL")
database_user = os.getenv("DATABASE_USER")
database_password = os.getenv("DATABASE_PASSWORD")

"""
host:服务器ip地址，默认为'localhost'
http_port:http协议——服务器监听端口，默认7474
https_port:https协议——服务器监听端口，默认7473
user:登录用户名，默认'neo4j'
password:登录密码，无默认值，故若数据库其他参数都为默认值，则可直接通过密码登录
"""
def connect():
    graph = Graph(database_url, auth=(database_user, database_password), name="neo4j")
    return graph

# 图
graph = connect()
# 匹配节点的匹配器
node_matcher = NodeMatcher(graph)
# 匹配相关边的匹配器
relationship_matcher = RelationshipMatcher(graph)
